package net.scribblemedia.candle.mlengine.training;

import org.encog.engine.network.activation.ActivationCompetitive;
import org.encog.engine.network.activation.ActivationLOG;
import org.encog.engine.network.activation.ActivationSigmoid;
import org.encog.engine.network.activation.ActivationTANH;

public enum ActivationFunctionType {
	
	ACTIVATION_SIGMOID(ActivationSigmoid.class), ACTIVATION_LOG(ActivationLOG.class),
	ACTIVATION_COMPETITIVE(ActivationCompetitive.class), ACTIVATION_TANH(ActivationTANH.class);
	
	@SuppressWarnings("rawtypes")
	private Class clazz;
	
	@SuppressWarnings("rawtypes")
	ActivationFunctionType(Class clazz) {
		this.clazz = clazz;
	}

	@SuppressWarnings("rawtypes")
	public Class getClazz() {
		return clazz;
	}
}
